import argparse
from predict import predict_model

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    default_str = 'default: %(default)s'
    parser.add_argument('--batchsize', type=int, default=16, help=default_str)
    parser.add_argument('--crop_shape', type=int, nargs=3, default=[64, 64, 64], help=default_str)
    parser.add_argument('--weight_decay', type=float, default=0, help=default_str)
    parser.add_argument('--prob_thresh', type=float, default=0.1, help=default_str)
    parser.add_argument('--train_or_test', type=str, default='train', help=default_str)
    parser.add_argument('--candidate_file', type=str, default='3dcnn_kaggle_testset_candidates_v4_newlabel.pkl', help=default_str)
    parser.add_argument('--candidate_vol_file', type=str, default='lidc_kaggle_candidates_v9.hdf5', help=default_str)
    parser.add_argument('--model_name', type=str, default='vgg_shortcut_1vs1.5', help=default_str)
    parser.add_argument('--model_epoch', type=str, default='epoch-0099', help=default_str)
    parser.add_argument('--gpus', type=str, default='0', help=default_str)
    args = parser.parse_args()
    predict_model(**vars(args))
